// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_RENDERER_MEDIA_MOCK_MEDIA_STREAM_VIDEO_SINK_H_
#define CONTENT_RENDERER_MEDIA_MOCK_MEDIA_STREAM_VIDEO_SINK_H_

#include "content/public/renderer/media_stream_video_sink.h"

#include "base/memory/weak_ptr.h"
#include "content/common/media/video_capture.h"
#include "media/base/video_frame.h"
#include "testing/gmock/include/gmock/gmock.h"

namespace content {

class MockMediaStreamVideoSink : public MediaStreamVideoSink {
public:
    MockMediaStreamVideoSink();
    ~MockMediaStreamVideoSink() override;

    void ConnectToTrack(const blink::WebMediaStreamTrack& track)
    {
        MediaStreamVideoSink::ConnectToTrack(track, GetDeliverFrameCB(), true);
    }

    void ConnectToTrackWithCallback(const blink::WebMediaStreamTrack& track,
        const VideoCaptureDeliverFrameCB& callback)
    {
        MediaStreamVideoSink::ConnectToTrack(track, callback, true);
    }

    void DisconnectFromTrack()
    {
        MediaStreamVideoSink::DisconnectFromTrack();
    }

    void OnReadyStateChanged(
        blink::WebMediaStreamSource::ReadyState state) override;
    void OnEnabledChanged(bool enabled) override;

    // Triggered when OnVideoFrame(const scoped_refptr<media::VideoFrame>& frame)
    // is called.
    MOCK_METHOD0(OnVideoFrame, void());

    VideoCaptureDeliverFrameCB GetDeliverFrameCB();

    int number_of_frames() const { return number_of_frames_; }
    media::VideoPixelFormat format() const { return format_; }
    gfx::Size frame_size() const { return frame_size_; }
    scoped_refptr<media::VideoFrame> last_frame() const { return last_frame_; };

    bool enabled() const { return enabled_; }
    blink::WebMediaStreamSource::ReadyState state() const { return state_; }

private:
    void DeliverVideoFrame(const scoped_refptr<media::VideoFrame>& frame,
        base::TimeTicks estimated_capture_time);

    int number_of_frames_;
    bool enabled_;
    media::VideoPixelFormat format_;
    blink::WebMediaStreamSource::ReadyState state_;
    gfx::Size frame_size_;
    scoped_refptr<media::VideoFrame> last_frame_;
    base::WeakPtrFactory<MockMediaStreamVideoSink> weak_factory_;
};

} // namespace content

#endif // CONTENT_RENDERER_MEDIA_MOCK_MEDIA_STREAM_VIDEO_SINK_H_
